# db_pool.py

from dbutils.pooled_db import PooledDB
import pymysql

# 创建 PyMySQL 连接池
POOL = PooledDB(
    creator=pymysql,  # 使用 PyMySQL 作为数据库驱动
    maxconnections=800,  # 最大连接数
    mincached=10,       # 初始化时创建的空闲连接数
    maxcached=50,       # 最大空闲连接数
    blocking=True,      # 连接池耗尽时是否阻塞等待
    host='127.0.0.1',   # 数据库主机
    user='root',        # 数据库用户
    password='123456',  # 数据库密码
    database='seckill_py',  # 数据库名称
    charset='utf8mb4'   # 字符集
)

# 获取连接
def get_connection():
    return POOL.connection()